import torch

torch.random.manual_seed(10)
data = torch.randint(0, 10, [3, 4, 5])
print(data)
# print(data[2,3])

# print(data[[0, 1], [1, 2]])
# print(data[[[1], [3]], [1, 2]])

# print(data[0:3:2, :2])
# print(data[2:, :2])

# print(data[data[:, 2] < 2 , :])
# print(data[:, 2] < 2)
# print(data[:, data[1] > 5])

# print(data[0, :, :])
# print(data[:, 0, :])
# print(data[:, :, 2])
print(data[1, 3, 4])
